package com.jiangyg.mall.core.utils;

import java.sql.Timestamp;
import java.text.ParseException;
import java.text.SimpleDateFormat;

/**
 * 工具描述：身份证验证辅助类
 *
 * @author jiangyg
 */
public final class IDCardUtils {

    private IDCardUtils() {
    }

    /**
     * 11个校验码字符
     */
    private static final char[] CODE = {'1', '0', 'X', '9', '8', '7', '6', '5', '4', '3', '2'};

    /**
     * 18个加权因子
     */
    private static final int[] FACTOR = {7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2, 1};

    /**
     * 功能描述：修补15位居民身份证号码为18位，并校验15位身份证有效性
     *
     * @param personIDCode 十五位身份证号码
     * @return String 十八位身份证号码
     */
    public static String fixPersonIDCodeWithCheck(String personIDCode) {
        if (personIDCode == null || personIDCode.trim().length() != 15)
            throw new RuntimeException("输入的身份证号不足15位，请检查");
        if (!isIdentity(personIDCode))
            throw new RuntimeException("输入的身份证号无效，请检查");
        return fixPersonIDCodeWithoutCheck(personIDCode);
    }

    /**
     * 功能描述：修补15位居民身份证号码为18位，不校验身份证有效性
     *
     * @param personIDCode 十五位身份证号码
     * @return 十八位身份证号码
     */
    public static String fixPersonIDCodeWithoutCheck(String personIDCode) {
        if (personIDCode == null || personIDCode.trim().length() != 15)
            throw new RuntimeException("输入的身份证号不足15位，请检查");
        String id17 = personIDCode.substring(0, 6) + "19" + personIDCode.substring(6, 15); // 15位身份证补'19'
        int[] idcd = new int[18];
        int sum; // 根据公式 ∑(ai×Wi) 计算
        int remainder; // 第18位校验码
        for (int i = 0; i < 17; i++) {
            idcd[i] = Integer.parseInt(id17.substring(i, i + 1));
        }
        sum = 0;
        for (int i = 0; i < 17; i++) {
            sum = sum + idcd[i] * FACTOR[i];
        }
        remainder = sum % 11;
        String lastCheckBit = String.valueOf(CODE[remainder]);
        return id17 + lastCheckBit;
    }

    /**
     * 功能描述：从身份证号中获取出生日期，身份证号可以为15位或18位
     *
     * @param identity 身份证号
     * @return 出生日期
     */
    public static Timestamp getBirthdayFromPersonIDCode(String identity) {
        String id = getFixedPersonIDCode(identity);
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        try {
            return new Timestamp(sdf.parse(id.substring(6, 14)).getTime());
        } catch (ParseException e) {
            throw new RuntimeException("不是有效的身份证号，请检查");
        }
    }

    /**
     * 功能描述：将传入的身份证号码进行校验，并返回一个对应的18位身份证
     *
     * @param personIDCode 身份证号码
     * @return String 十八位身份证号码
     */
    public static String getFixedPersonIDCode(String personIDCode) {
        if (personIDCode == null)
            throw new RuntimeException("输入的身份证号无效，请检查");
        if (personIDCode.length() == 18) {
            if (isIdentity(personIDCode))
                return personIDCode;
            else
                throw new RuntimeException("输入的身份证号无效，请检查");
        } else if (personIDCode.length() == 15)
            return fixPersonIDCodeWithCheck(personIDCode);
        else
            throw new RuntimeException("输入的身份证号无效，请检查");
    }

    /**
     * 功能描述：从身份证号获取性别
     *
     * @param identity 身份证号
     * @return 性别代码 1-女 2-男
     */
    public static String getGenderFromPersonIDCode(String identity) {
        String id = getFixedPersonIDCode(identity);
        char sex = id.charAt(16);
        return sex % 2 == 0 ? "1" : "2";
    }

    /**
     * 功能描述：判断是否是有效的18位或15位居民身份证号码
     *
     * @param identity 18位或15位居民身份证号码
     * @return 是否为有效的身份证号码
     */
    public static boolean isIdentity(String identity) {
        if (identity == null)
            return false;
        if (identity.length() == 18 || identity.length() == 15) {
            String id15;
            if (identity.length() == 18)
                id15 = identity.substring(0, 6) + identity.substring(8, 17);
            else
                id15 = identity;
            try {
                Long.parseLong(id15); // 校验是否为数字字符串

                String birthday = "19" + id15.substring(6, 12);
                SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
                sdf.parse(birthday); // 校验出生日期
                if (identity.length() == 18 && !fixPersonIDCodeWithoutCheck(id15).equals(identity))
                    return false; // 校验18位身份证
            } catch (Exception e) {
                return false;
            }
            return true;
        } else
            return false;
    }

}
